613b414c1c6ad328d8ba0d2a1107ef617582143a,python/src/com/jetbrains/python/psi/impl/PyTargetExpressionImpl.java,PyTargetExpressionImpl,getType,#TypeEvalContext#,100

Before Change


            if (cls != null) {
              final PyFunction enter = cls.findMethodByName(PyNames.ENTER, true);
              if (enter != null) {
                return enter.getReturnType(context, null);
              }
            }
          }

After Change


            if (cls != null) {
              final PyFunction enter = cls.findMethodByName(PyNames.ENTER, true);
              if (enter != null) {
                final PyType enterType = enter.getReturnType(context, null);
                if (enterType != null) {
                  return enterType;
                }
                // Guess the return type of __enter__
                return PyUnionType.createWeakType(exprType);
              }
            }
          }